from dotenv import load_dotenv, find_dotenv
import os
import Agently

_ = load_dotenv(find_dotenv())

# 创建一个Agent工厂实例
agent_factory = (
    Agently.AgentFactory()
        # 给Agent工厂实例提供设置项：
        .set_settings("current_model", "OpenAI")
        .set_settings("model.OpenAI.auth", { "api_key": os.getenv("OPENAI_API_KEY") })
        # 如果需要切换模型，可参考模型清单: https://platform.openai.com/docs/models/overview
        # 默认模型: gpt-3.5-turbo
        .set_settings("model.OpenAI.options", { "model": "gpt-4o" })
        # 如果您需要使用转发Base URL，通常为https://base_domain_name.com/v1这样的格式
        .set_settings("model.OpenAI.url", os.getenv("OPENAI_API_BASE"))
        # 如果您需要使用本地代理，确定本地代理的host和port后，可以按如下方式设置
        # .set_settings("proxy", "http://127.0.0.1:7890")
        # .set_settings("current_model", "Google")
        # .set_settings("model.Google.auth", { "api_key": os.getenv("GOOGLE_API_KEY") })
)

if __name__ == "__main__":
    agent = agent_factory.create_agent()
    result = agent.input(input("[请输入您的要求]: ")).start()
    print("[回复]: ", result)